23-H0ngJu #227

H0ngJu · 2024-08-10T11:36:47Z

🔗 문제 링크

베스트앨범

✔️ 소요된 시간

45M

✨ 수도 코드

문제 요약
재생수가 가장 많은 장르 2개를 골라, 각각의 장르에서 2개씩 선택하여 베스트 앨범을 만들어라@!

저번 문제에 이은 해시 문제입니다.

문제에서 요구하는대로 차근차근 풀면 큰 어려움 없이 풀 수 있어요

일단 가장 먼저 알아야 할 정보는 총 재생 수가 많은 장르 2개입니다.

따라서 먼저,
(인덱스, 장르)를 알려줄 딕셔너리인 info를 하나 생성하고,
(재생수 합, 장르)의 정보를 담을 딕셔너리 dic을 하나 생성합니다.

dic = {}
info = {}

for i in range(len(genres)):
    g = genres[i]
    p = plays[i]

    info[i] = g #인덱스 저장
    
    if g in dic:
        dic[g] += p #총 재생수 저장
    else:
        dic[g] = p

그리고 완성된 dic에서 value값을 기준으로 내림차순으로 정렬한 리스트 best_g를 하나 생성합니다.
그러면 best_g에는 재생수를 기준으로 하여 내림차순으로 정렬된 (장르, 재생수)가 저장됩니다.

best_g = sorted(dic.items(), key=lambda x: x[1], reverse=True)

이제, best_g에서 하나씩 장르명을 꺼내줍니다.

그리고 그 장르명에 대해서 genre(처음에 주어진 input)와 비교하면서 같으면 best_album에 담습니다.

이때, 해당 장르 내에서 다시 재생 수가 많은 2곡을 뽑아야 하므로,
그냥 버블 정렬로 내림차순으로 정렬해줍니다.

그러고 나서 best_album의 1,2번째 원소만 answer에 추가해주면 끝이 납니다.

for g, _ in best_g:
    if f == 2: break
    best_album = [(plays[i], i) for i in range(len(genres)) if genres[i] == g]
    for i in range(len(best_album)):
        for j in range(i+1, len(best_album)):
            if best_album[j][0] > best_album[i][0]: #재생수가 작은 경우
                best_album[i], best_album[j] = best_album[j], best_album[i]
    
    answer.extend([x[1] for x in best_album[:2]])
    f += 1

사실 처음 코드에는 같은 재생수일 때는 인덱스가 작은 곡이 먼저 오도록 했는데,,,
지금 다시 보니까 해당 조건이 없네요 허ㅓㅎ;; 이 문제 풀기 전에 푼 문제랑 짬뽕됐나 봅니둥

문제를 쫌 더 제대로 읽어야겠습니다....ㅎ

📚 새롭게 알게된 내용

tgyuuAn

이 문제도 마찬가지로 예전에 풀었었는데,

풀이가 바로 떠올라서 코드를 다시 치진 않았습니다!!!

def solution(genres, plays):
    dict = {}
    dict_max = {}
    answer = []
    for genre in range(len(genres)):
        if genres[genre] not in dict:
            dict[genres[genre]] = [[genre,plays[genre]]]
        else:
            dict[genres[genre]].append([genre,plays[genre]])

    for genre in dict:
        tot =0
        for j in dict[genre]:
            tot += j[1]
        dict_max[genre]=tot
        dict[genre].sort(reverse=True, key=lambda x: x[1])
    
    dict_max = sorted(list(dict_max.items()), reverse = True, key = lambda x:x[1])
    
    for genre in dict_max:
        count = 0
        while dict[genre[0]]:
            if len(dict[genre[0]])==0 or count ==2:
                break
            else:
                answer.append(dict[genre[0]].pop(0)[0])
                count +=1
    return answer

해시에다가 정렬에다가 구현에다가..

문제 자체의 난이도는 쉬운데 여러가지를 요하네요..!

처음에 연습할 때 짱 좋은 문제인듯..!

InSange

파이썬이 확실히 코드상으로는 가벼워보이는 감이 있긴하군요..
저는 이거 풀었을 때 조금 고생했던 것 같습니다.
장르별로 인덱스와 값에 대한 데이터를 위한 해시 map
각각의 장르를 중복이 허용되지 않는 집합을 통해 각 장르가 한 번씩 접근해주기 위한 set
그리고 데이터 정렬을 위한 우선 순위 큐를 활용해서 풀었습니다..

#include <string>
#include <vector>
#include <unordered_map>
#include <unordered_set>
#include <queue>
#include <iostream>

using namespace std;

struct cmp{
    bool operator()(const pair<int, int>& a, const pair<int, int>& b)
    {
        if(a.first == b.first) return a.second > b.second;
        return a.first < b.first;
    }
};

vector<int> solution(vector<string> genres, vector<int> plays) {
    vector<int> answer;
    unordered_map<string, int> album_data;
    unordered_map<string, priority_queue<pair<int, int>, vector<pair<int, int>>, cmp>> album_index;
    unordered_set<string> genre_data;
    priority_queue<pair<int, string>> pq;
    
    for(int i = 0; i < genres.size(); i++)
    {
        string genre = genres[i];
        int play = plays[i];
        
        album_data[genre] += play;
        album_index[genre].push({play, i});
        genre_data.insert(genre);
    }
    
    for(unordered_set<string>::const_iterator it = genre_data.begin(); it != genre_data.end(); ++it)
    {
        pq.push({album_data[*it], *it});
    }
    
    while(!pq.empty())
    {
        int cnt = 0;
        string genre = pq.top().second;
        while(cnt < 2 && !album_index[genre].empty())
        {
            answer.push_back(album_index[genre].top().second);
            album_index[genre].pop();
            cnt++;
        }
        pq.pop();
    }
    
    return answer;
}

9kyo-hwang

예전에

[장르 : <총 재생 횟수, <재생 횟수, 고유 번호> 리스트>] 딕셔너리를 만들어서 기록
딕셔너리의 <총 재생 횟수, <재생 횟수, 고유 번호> 리스트> Value 리스트를 추출해서 총 재생 횟수 기준으로 내림차순 정렬
추출한 리스트를 순회하며 <총 재생 횟수, <재생 횟수, 고유 번호> 리스트>의 <재생 횟수, 고유 번호> 리스트를 재생 횟수 내림차순 -> 고유 번호 오름차순으로 정렬
정렬한 리스트에서 2개씩 Id 추출

식으로 풀었었네요... 뭔가 복잡합니다.

전체 코드

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>

using namespace std;

vector<int> solution(vector<string> Genres, vector<int> Plays) 
{
    using FSongs = pair<int, vector<pair<int, int>>>;
    unordered_map<string, FSongs> Albums;
    
    for(int i = 0; i < Genres.size(); ++i)
    {
        auto& [TotalPlay, Songs] = Albums[Genres[i]];
        
        TotalPlay += Plays[i];
        Songs.emplace_back(Plays[i], i);
    }
    
    vector<FSongs> SongsList(Albums.size());
    transform(Albums.begin(), Albums.end(), SongsList.begin(), 
              [](auto const& Album)
              {
                  return Album.second;
              });
    sort(SongsList.begin(), SongsList.end(), greater<>());  // TotalPlay 내림차순
    
    vector<int> Answer;
    for(auto& [TotalPlay, Songs] : SongsList)
    {
        sort(Songs.begin(), Songs.end(), [](auto const& Lhs, auto const& Rhs)
             {
                 return Lhs.first == Rhs.first ? Lhs.second < Rhs.second : Lhs.first > Rhs.first;
             });
        
        int Count = 0;
        for(const auto& [Play, Id] : Songs)
        {
            if(Count == 2) break;
            
            Answer.emplace_back(Id);
            Count++;
        }
    }
    
    return Answer;
}

이번에 다시 풀어봤는데,

[장르, 재생 횟수, 고유 번호]를 가지는 FAlbum 클래스 생성. 재생 횟수 기준으로 정렬되게 설정
[장르 : 총 재생 횟수] 딕셔너리, FAlbum 리스트 2개를 만들어 각각 기록
[장르 : 총 재생 횟수] 딕셔너리를 리스트로 변환, 총 재생 횟수 내림차순으로 정렬
FAlbum 리스트를 내림차순(재생 횟수 기준) 정렬
정렬된 장르 별 총 재생횟수 리스트와 FAlbum 리스트를 순회해 각 장르 별 노래 고유 번호 2개씩 저장

식으로도 풀리네요. 조금? 코드가 간결해진 것 같기도 하구요...

전체 코드

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>

using namespace std;

struct FAlbum
{
    string Genre;
    int Play;
    int Id;
    
    FAlbum(string InGenre, int InPlay, int InId)
        : Genre(std::move(InGenre))
        , Play(InPlay)
        , Id(InId)
        {}
    
    bool operator<(const FAlbum& Other) const
    {
        return Play < Other.Play;
    }
    bool operator<=(const FAlbum& Other) const
    {
        return Play <= Other.Play;
    }
    bool operator>(const FAlbum& Other) const
    {
        return Play > Other.Play;
    }
    bool operator>=(const FAlbum& Other) const
    {
        return Play >= Other.Play;
    }
    bool operator==(const FAlbum& Other) const
    {
        return Play == Other.Play;
    }
    bool operator!=(const FAlbum& Other) const
    {
        return Play != Other.Play;
    }
};

vector<int> solution(vector<string> Genres, vector<int> Plays) 
{
    vector<FAlbum> Albums;
    unordered_map<string, int> PlaysbyGenre;
    for(int i = 0; i < Genres.size(); ++i)
    {
        PlaysbyGenre[Genres[i]] += Plays[i];
        Albums.emplace_back(Genres[i], Plays[i], i);
    }
    
    vector<pair<string, int>> PlaysbyGenreList(PlaysbyGenre.begin(), PlaysbyGenre.end());
    
    sort(PlaysbyGenreList.begin(), PlaysbyGenreList.end(), 
         [](auto const& Lhs, auto const& Rhs)
         {
             return Lhs.second > Rhs.second;
         });
    
    sort(Albums.begin(), Albums.end(), greater<>());
    
    vector<int> Answer;
    for(const auto& [Genre, Play] : PlaysbyGenreList)
    {
        int Count = 0;
        for(const auto& Album : Albums)
        {
            if(Genre == Album.Genre)
            {
                Count++;
                Answer.emplace_back(Album.Id);
            }
            
            if(Count == 2)
            {
                break;
            }
        }
    }
    
    return Answer;
}

2024-08-10

2ee6c1e

H0ngJu added 작성 중 ⏱️ H0ngJu labels Aug 10, 2024

H0ngJu self-assigned this Aug 10, 2024

2024-08-10

9c598c7

H0ngJu requested review from InSange, yuyu0830, seongwon030 and tgyuuAn August 10, 2024 12:12

H0ngJu added 리뷰 기다리는 중 🔥 and removed 작성 중 ⏱️ labels Aug 10, 2024

H0ngJu changed the title ~~2024-08-10~~ 23-H0ngJu Aug 10, 2024

tgyuuAn approved these changes Aug 10, 2024

View reviewed changes

InSange approved these changes Aug 11, 2024

View reviewed changes

9kyo-hwang approved these changes Aug 14, 2024

View reviewed changes

tgyuuAn removed request for yuyu0830 and seongwon030 August 20, 2024 15:22

tgyuuAn added 리뷰 완료 ✔️ and removed 리뷰 기다리는 중 🔥 labels Aug 20, 2024

Merge branch 'main' into 23-H0ngJu

2dd22df

H0ngJu merged commit 81f4b15 into main Aug 21, 2024

H0ngJu deleted the 23-H0ngJu branch August 21, 2024 03:44

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

23-H0ngJu #227

23-H0ngJu #227

H0ngJu commented Aug 10, 2024 •

edited

Loading

tgyuuAn left a comment

InSange left a comment

9kyo-hwang left a comment

23-H0ngJu #227

23-H0ngJu #227

Conversation

H0ngJu commented Aug 10, 2024 • edited Loading

🔗 문제 링크

✔️ 소요된 시간

✨ 수도 코드

📚 새롭게 알게된 내용

tgyuuAn left a comment

Choose a reason for hiding this comment

InSange left a comment

Choose a reason for hiding this comment

9kyo-hwang left a comment

Choose a reason for hiding this comment

H0ngJu commented Aug 10, 2024 •

edited

Loading